pd_covid_loc = pd_covid[[
'location',
'date',
'total_cases',
'life_expectancy',
'days_since_first',
'days_since_tenth',
'days_since_hundreth',
]]
pd_covid_loc.loc[:, ('prediction')] = pd_covid.loc[:, ('ml.total_cases_prediction')]
pd_covid_loc.loc[:, ('life_expectancy_importance')] = pd_covid.loc[:, ('ml.feature_importance.life_expectancy')]
pd_covid_loc.loc[:, ('dalys_importance')] = pd_covid.loc[:, ('ml.feature_importance.dalys')]
pd_covid_loc.loc[:, ('days_since_first_importance')] = pd_covid.loc[:, ('ml.feature_importance.days_since_first')]
pd_covid_loc.loc[:, ('days_since_tenth_importance')] = pd_covid.loc[:, ('ml.feature_importance.days_since_tenth')]
pd_covid_loc.loc[:, ('days_since_hundreth_importance')] = pd_covid.loc[:, ('ml.feature_importance.days_since_hundreth')]
pd_covid_loc.loc[:, ('population_importance')] = pd_covid.loc[:, ('ml.feature_importance.population')]
pd_covid_loc.loc[:, ('age_young_importance')] = pd_covid.loc[:, ('ml.feature_importance.age_young')]
pd_covid_loc.loc[:, ('age_working_importance')] = pd_covid.loc[:, ('ml.feature_importance.age_working')]
pd_covid_loc.loc[:, ('age_eldery_importance')] = pd_covid.loc[:, ('ml.feature_importance.age_eldery')]
alt.data_transformers.disable_max_rows()
chart_cases = alt.Chart(pd_covid_loc).mark_line(
color='blue',
opacity=0.5,
size=2
).encode(
alt.X('date:T', title=''),
alt.Y('total_cases:Q', title='', scale=alt.Scale(domain=[0, 900000])),
tooltip=[alt.Tooltip('date:T', title='Date'), alt.Tooltip('total_cases:Q', title='Total Cases')]
)
chart_cases_predicted = alt.Chart(pd_covid_loc).mark_line(
color='red',
opacity=0.5,
size=2
).encode(
alt.X('date:T', title=''),
alt.Y('prediction:Q', title='', scale=alt.Scale(domain=[0, 10000])),
tooltip=[alt.Tooltip('date:T', title='Date'), alt.Tooltip('total_cases:Q', title='Total Cases Prediction')]
)
(chart_cases + chart_cases_predicted).properties(
width=100,
height=120
).facet(
facet='location:N',
columns=5
)